from pydantic import BaseModel


class NodeInfo(BaseModel):
    labels: dict
    node_name: str
    gpu: int
    docker_version: str
    kernel_version: str
    os_image: str
    ip: str
